In [1]:
# Author: Luke McEvoy
# Contact: lukemcevoy7@hotmail.com, 631-478-9254
# OKC Data Science Take Home Summer 2022
In [2]:
# Import Libraries
import pandas as pd
import seaborn as sns
import math
import matplotlib.pyplot as plt
from matplotlib.patches import Circle, Rectangle, Arc
In [3]:
# Read data on Team A and B shooting
okc_df = pd.read_csv('/Users/lukemcevoy/Develop/self/work/interviews/okc/nba-data-visualization/shots_data.csv')
In [4]:
# Plot Team A and B's heatmap of all shots (made and missed)
cmap = plt.cm.Greens_r
joint_shot_chart = sns.jointplot(x=okc_df['x'], y=okc_df['y'], kind='kde', space=0, color=cmap(0.1), cmap=cmap, n_levels=50)
joint_shot_chart.fig.set_size_inches(12,10)
ax = joint_shot_chart.ax_joint
ax.set_xlim(-30,30)
ax.set_ylim(-5,35)

plt.title('All Shots from Team A and B')

plt.show()
In [5]:
okc_df_made = okc_df[okc_df['fgmade']==1]
okc_df_miss = okc_df[okc_df['fgmade']==0]
In [6]:
# Plot Team A and B's heatmap of all made shots
cmap = plt.cm.Greens_r
joint_shot_chart = sns.jointplot(x=okc_df_made['x'], y=okc_df_made['y'], kind='kde', space=0, color=cmap(0.1), cmap=cmap, n_levels=50)
joint_shot_chart.fig.set_size_inches(12,10)
ax = joint_shot_chart.ax_joint
ax.set_xlim(-30,30)
ax.set_ylim(-5,35)

plt.title('All Made Shots from Team A and B')

plt.show()
In [7]:
# Plot Team A and B's heatmap of all missed shots
cmap = plt.cm.Greens_r
joint_shot_chart = sns.jointplot(x=okc_df_miss['x'], y=okc_df_miss['y'], kind='kde', space=0, color=cmap(0.1), cmap=cmap, n_levels=50)
joint_shot_chart.fig.set_size_inches(12,10)
ax = joint_shot_chart.ax_joint
ax.set_xlim(-30,30)
ax.set_ylim(-5,35)

plt.title('All Missed Shots from Team A and B')

plt.show()
In [8]:
# Split read dataframe (df) on team. Two dfs, one of Team A one of Team B
mask = okc_df['team'] == 'Team A'
team_a = okc_df[mask]
team_b = okc_df[~mask]
In [9]:
cmap = plt.cm.Reds_r
joint_shot_chart = sns.jointplot(x=team_a['x'], y=team_a['y'], kind='kde', space=0, color=cmap(0.1), cmap=cmap, n_levels=50)
joint_shot_chart.fig.set_size_inches(12,10)
ax = joint_shot_chart.ax_joint
ax.set_xlim(-30,30)
ax.set_ylim(-5,35)

plt.title('All Shots from Team A')

plt.show()
In [10]:
team_a_made = team_a[team_a['fgmade']==1]
team_a_miss = team_a[team_a['fgmade']==0]
In [11]:
cmap = plt.cm.Reds_r
joint_shot_chart = sns.jointplot(x=team_a_made['x'], y=team_a_made['y'], kind='kde', space=0, color=cmap(0.1), cmap=cmap, n_levels=50)
joint_shot_chart.fig.set_size_inches(12,10)
ax = joint_shot_chart.ax_joint
ax.set_xlim(-30,30)
ax.set_ylim(-5,35)

plt.title('All Made Shots from Team A')

plt.show()
In [12]:
cmap = plt.cm.Reds_r
joint_shot_chart = sns.jointplot(x=team_a_miss['x'], y=team_a_miss['y'], kind='kde', space=0, color=cmap(0.1), cmap=cmap, n_levels=50)
joint_shot_chart.fig.set_size_inches(12,10)
ax = joint_shot_chart.ax_joint
ax.set_xlim(-30,30)
ax.set_ylim(-5,35)

plt.title('All Missed Shots from Team A')

plt.show()
In [13]:
cmap = plt.cm.Blues_r
joint_shot_chart = sns.jointplot(x=team_b['x'], y=team_b['y'], kind='kde', space=0, color=cmap(0.1), cmap=cmap, n_levels=50)
joint_shot_chart.fig.set_size_inches(12,10)
ax = joint_shot_chart.ax_joint
ax.set_xlim(-30,30)
ax.set_ylim(-5,35)

plt.title('All Shots from Team B')

plt.show()
In [14]:
team_b_made = team_b[team_b['fgmade']==1]
team_b_miss = team_b[team_b['fgmade']==0]
In [15]:
cmap = plt.cm.Reds_r
joint_shot_chart = sns.jointplot(x=team_b_made['x'], y=team_b_made['y'], kind='kde', space=0, color=cmap(0.1), cmap=cmap, n_levels=50)
joint_shot_chart.fig.set_size_inches(12,10)
ax = joint_shot_chart.ax_joint
ax.set_xlim(-30,30)
ax.set_ylim(-5,35)

plt.title('All Made Shots from Team B')

plt.show()
In [16]:
cmap = plt.cm.Reds_r
joint_shot_chart = sns.jointplot(x=team_b_miss['x'], y=team_b_miss['y'], kind='kde', space=0, color=cmap(0.1), cmap=cmap, n_levels=50)
joint_shot_chart.fig.set_size_inches(12,10)
ax = joint_shot_chart.ax_joint
ax.set_xlim(-30,30)
ax.set_ylim(-5,35)

plt.title('All Missed Shots from Team B')

plt.show()
In [17]:
plt.figure(figsize=(12, 10))
plt.scatter(okc_df['x'], okc_df['y'], c=okc_df['fgmade'], cmap='coolwarm_r')
cur_axes = plt.gca()
plt.xlim([-30, 30])
plt.ylim([-5, 35])
plt.title("All Shots (Blue made, Red missed) from Team A and B")
plt.show()
In [18]:
plt.figure(figsize=(12, 10))
plt.scatter(team_a['x'], team_a['y'], c=team_a['fgmade'], cmap='coolwarm_r')
cur_axes = plt.gca()
plt.xlim([-30, 30])
plt.ylim([-5, 35])
plt.title("All Shots (Blue made, Red missed) from Team A")
plt.show()
In [19]:
plt.figure(figsize=(12, 10))
plt.scatter(team_b['x'], team_b['y'], c=team_b['fgmade'], cmap='coolwarm_r')
cur_axes = plt.gca()
plt.xlim([-30, 30])
plt.ylim([-5, 35])
plt.title("All Shots (Blue made, Red missed) from Team B")
plt.show()
In [20]:
team_a_corner_3s = team_a.loc[(team_a['y'] <= 7.8) & ((team_a['x'] > 22) | (team_a['x'] < -22))]
In [21]:
cmap = plt.cm.Reds_r
joint_shot_chart = sns.jointplot(x=team_a_corner_3s['x'], y=team_a_corner_3s['y'], kind='kde', space=0, color=cmap(0.1), cmap=cmap, n_levels=50)
joint_shot_chart.fig.set_size_inches(12,10)
ax = joint_shot_chart.ax_joint
ax.set_xlim(-30,30)
ax.set_ylim(-2,7.8)

plt.title('Corner 3 Shots from Team A')

plt.show()
In [22]:
team_b_corner_3s = team_b.loc[(team_b['y'] <= 7.8) & ((team_b['x'] > 22) | (team_b['x'] < -22))]
In [23]:
cmap = plt.cm.Reds_r
joint_shot_chart = sns.jointplot(x=team_b_corner_3s['x'], y=team_b_corner_3s['y'], kind='kde', space=0, color=cmap(0.1), cmap=cmap, n_levels=50)
joint_shot_chart.fig.set_size_inches(12,10)
ax = joint_shot_chart.ax_joint
ax.set_xlim(-30,30)
ax.set_ylim(-3,7.8)

plt.title('Corner 3 Shots from Team B')

plt.show()
In [24]:
plt.figure(figsize=(12, 10))
plt.scatter(team_a_corner_3s['x'], team_a_corner_3s['y'], c=team_a_corner_3s['fgmade'], cmap='coolwarm_r')
cur_axes = plt.gca()
plt.xlim([-30, 30])
plt.ylim([0, 7.8])
plt.title("All Corner 3 Shots (Blue made, Red missed) from Team A")
plt.show()
In [25]:
plt.figure(figsize=(12, 10))
plt.scatter(team_b_corner_3s['x'], team_b_corner_3s['y'], c=team_b_corner_3s['fgmade'], cmap='coolwarm_r')
cur_axes = plt.gca()
plt.xlim([-30, 30])
plt.ylim([0, 7.8])
plt.title("All Corner 3 Shots (Blue made, Red missed) from Team B")
plt.show()
In [26]:
team_a_non_corner_3s = team_a.loc[(team_a['y'] > 7.8)]
In [27]:
for index, row in team_a_non_corner_3s.iterrows():
    if math.hypot(row['x'], row['y']) < 23.75:
        team_a_non_corner_3s = team_a_non_corner_3s.drop(index)
In [28]:
cmap = plt.cm.Blues_r
joint_shot_chart = sns.jointplot(x=team_a_non_corner_3s['x'], y=team_a_non_corner_3s['y'], kind='kde', space=0, color=cmap(0.1), cmap=cmap, n_levels=50)
joint_shot_chart.fig.set_size_inches(12,10)
ax = joint_shot_chart.ax_joint
ax.set_xlim(-30,30)
ax.set_ylim(7,35)

plt.title('All Non Corner 3s from Team A')

plt.show()
In [29]:
team_b_non_corner_3s = team_b.loc[(team_b['y'] > 7.8)]
In [30]:
for index, row in team_b_non_corner_3s.iterrows():
    if math.hypot(row['x'], row['y']) < 23.75:
        team_b_non_corner_3s = team_b_non_corner_3s.drop(index)
In [31]:
cmap = plt.cm.Blues_r
joint_shot_chart = sns.jointplot(x=team_b_non_corner_3s['x'], y=team_b_non_corner_3s['y'], kind='kde', space=0, color=cmap(0.1), cmap=cmap, n_levels=50)
joint_shot_chart.fig.set_size_inches(12,10)
ax = joint_shot_chart.ax_joint
ax.set_xlim(-30,30)
ax.set_ylim(7,35)

plt.title('All Non Corner 3s from Team B')

plt.show()
In [32]:
plt.figure(figsize=(12, 10))
plt.scatter(team_a_non_corner_3s['x'], team_a_non_corner_3s['y'], c=team_a_non_corner_3s['fgmade'], cmap='coolwarm_r')
cur_axes = plt.gca()
plt.xlim([-30, 30])
plt.ylim([-5, 35])
plt.title("All Non Corner 3s (Blue made, Red missed) from Team A")
plt.show()
In [33]:
plt.figure(figsize=(12, 10))
plt.scatter(team_b_non_corner_3s['x'], team_b_non_corner_3s['y'], c=team_b_non_corner_3s['fgmade'], cmap='coolwarm_r')
cur_axes = plt.gca()
plt.xlim([-30, 30])
plt.ylim([-5, 35])
plt.title("All Non Corner 3s (Blue made, Red missed) from Team B")
plt.show()
In [34]:
team_a_two_pointers = []
for index, row in team_a.iterrows():
    if row['y'] <= 7.8 and (row['x'] < 22 or row['x'] > -22):
        team_a_two_pointers.append(row)
    if row['y'] > 7.8 and math.hypot(row['x'],row['y']) < 23.75:
            team_a_two_pointers.append(row)
team_a_two_pointers_df = pd.DataFrame(team_a_two_pointers)
In [35]:
cmap = plt.cm.Blues_r
joint_shot_chart = sns.jointplot(x=team_a_two_pointers_df['x'], y=team_a_two_pointers_df['y'], kind='kde', space=0, color=cmap(0.1), cmap=cmap, n_levels=50)
joint_shot_chart.fig.set_size_inches(12,10)
ax = joint_shot_chart.ax_joint
ax.set_xlim(-30,30)
ax.set_ylim(-5,35)

plt.title('All 2 Pointers from Team A')

plt.show()
In [36]:
team_b_two_pointers = []
for index, row in team_b.iterrows():
    if row['y'] <= 7.8 and (row['x'] < 22 or row['x'] > -22):
        team_b_two_pointers.append(row)
    if row['y'] > 7.8 and math.hypot(row['x'],row['y']) < 23.75:
            team_b_two_pointers.append(row)
team_b_two_pointers_df = pd.DataFrame(team_b_two_pointers)
In [37]:
cmap = plt.cm.Blues_r
joint_shot_chart = sns.jointplot(x=team_b_two_pointers_df['x'], y=team_b_two_pointers_df['y'], kind='kde', space=0, color=cmap(0.1), cmap=cmap, n_levels=50)
joint_shot_chart.fig.set_size_inches(12,10)
ax = joint_shot_chart.ax_joint
ax.set_xlim(-30,30)
ax.set_ylim(-5,35)

plt.title('All 2 Pointers from Team B')

plt.show()
In [38]:
plt.figure(figsize=(12, 10))
plt.scatter(team_a_two_pointers_df['x'], team_a_two_pointers_df['y'], c=team_a_two_pointers_df['fgmade'], cmap='coolwarm_r')
cur_axes = plt.gca()
plt.xlim([-30, 30])
plt.ylim([-5, 35])
plt.title("All Two Pointers (Blue made, Red missed) from Team A")
plt.show()
In [39]:
plt.figure(figsize=(12, 10))
plt.scatter(team_b_two_pointers_df['x'], team_b_two_pointers_df['y'], c=team_b_two_pointers_df['fgmade'], cmap='coolwarm_r')
cur_axes = plt.gca()
plt.xlim([-30, 30])
plt.ylim([-5, 35])
plt.title("All Two Pointers (Blue made, Red missed) from Team B")
plt.show()
In [40]:
def eFG(fgm, fga, threepm):
    return (fgm + (0.5 + threepm)) / fga
In [41]:
team_a_two_pointer_fgm = len(team_a_two_pointers_df[team_a_two_pointers_df['fgmade'] == 1])
team_a_two_pointer_fga = len(team_a_two_pointers_df)
In [42]:
print('Team A Two Pointer FG % = ', team_a_two_pointer_fgm / team_a_two_pointer_fga)

Team A Two Pointer FG % =  0.4
In [43]:
team_b_two_pointer_fgm = len(team_b_two_pointers_df[team_b_two_pointers_df['fgmade'] == 1])
team_b_two_pointer_fga = len(team_b_two_pointers_df)
In [44]:
print('Team B Two Pointer FG % = ', team_b_two_pointer_fgm / team_b_two_pointer_fga)

Team B Two Pointer FG % =  0.43478260869565216
In [45]:
team_a_corner_3s_fgm = len(team_a_corner_3s[team_a_corner_3s['fgmade'] == 1])
team_a_corner_3s_fga = len(team_a_corner_3s)
In [46]:
print('Team A Corner 3 FG % = ', team_a_corner_3s_fgm / team_a_corner_3s_fga)

Team A Corner 3 FG % =  0.4
In [47]:
team_b_corner_3s_fgm = len(team_b_corner_3s[team_b_corner_3s['fgmade'] == 1])
team_b_corner_3s_fga = len(team_b_corner_3s)
In [48]:
print('Team B Corner 3 FG % = ', team_b_corner_3s_fgm / team_b_corner_3s_fga)

Team B Corner 3 FG % =  0.23809523809523808
In [49]:
team_a_non_corner_3s_fgm = len(team_a_non_corner_3s[team_a_non_corner_3s['fgmade'] == 1])
team_a_non_corner_3s_fga = len(team_a_non_corner_3s)
In [50]:
print('Team A Non Corner 3 FG % = ', team_a_non_corner_3s_fgm / team_a_non_corner_3s_fga)

Team A Non Corner 3 FG % =  0.32222222222222224
In [51]:
team_b_non_corner_3s_fgm = len(team_b_non_corner_3s[team_b_non_corner_3s['fgmade'] == 1])
team_b_non_corner_3s_fga = len(team_b_non_corner_3s)
In [52]:
print('Team B Non Corner 3 FG % = ', team_b_non_corner_3s_fgm / team_b_non_corner_3s_fga)

Team B Non Corner 3 FG % =  0.3645833333333333
In [53]:
print('Team A EFG', eFG(team_a_non_corner_3s_fgm, team_a_non_corner_3s_fga, team_a_corner_3s_fgm + team_a_non_corner_3s_fgm))

Team A EFG 0.7388888888888889
In [54]:
print('Team B EFG', eFG(team_b_non_corner_3s_fgm, team_b_non_corner_3s_fga, team_b_corner_3s_fgm + team_b_non_corner_3s_fgm))

Team B EFG 0.7864583333333334
In [ ]:
In [ ]:
In [ ]:
In [55]:
# 5. What percentage of Team A's shots were attempted in the 2PT zone
team_a_two_pointer_fga / len(team_a)
Out[55]:
0.6785714285714286
In [56]:
# 6. What percentage of Team A's shots were attemped in non corner 3 zone
team_a_non_corner_3s_fga / len(team_a)
Out[56]:
0.32142857142857145
In [57]:
# 7. What percentage of Team A's shots were attempted in corner 3 zone
team_a_corner_3s_fga / len(team_a)
Out[57]:
0.07142857142857142
In [58]:
# 8910What was the eFG in the 2PT zone for Team A
eFG()

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-58-90cb3f28315d> in <module>
      1 # 8910What was the eFG in the 2PT zone for Team A
----> 2 eFG()

TypeError: eFG() missing 3 required positional arguments: 'fgm', 'fga', and 'threepm'
In [ ]:
In [ ]:
In [ ]:
In [ ]:
In [ ]: